ObjectDetection

sealed class ObjectDetection<U : InferenceModel> : OnnxModelType<U>

Object detection models and preprocessing.

Types

EfficientDetLite0

android

object EfficientDetLite0 : ONNXModels.ObjectDetection<SSDLikeModel>

This model is a real-time neural network for object detection that detects 90 different classes (labels are available in org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017).

SSDMobileNetV1

android

object SSDMobileNetV1 : ONNXModels.ObjectDetection<SSDLikeModel>

This model is a real-time neural network for object detection that detects 90 different classes (labels are available in org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017).

Functions

model

android

open fun model(modelHub: ModelHub): OnnxInferenceModel

pretrainedModel

android

abstract fun pretrainedModel(modelHub: ModelHub): U

Properties

inputShape

android

open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath

android

open override val modelRelativePath: String

preprocessor

android

open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Inheritors

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection